from typing import Any

from pydantic import BaseModel


class ResponseModel(BaseModel):
    data: Any
    code: str
    message: str


class UserBase(BaseModel):
    user_id: int
    username: str


class UserCreate(BaseModel):
    username: str
    password: str


class UserLogin(BaseModel):
    username: str
    password: str


class ConversationCreate(BaseModel):
    conversation_id: str


class MessageCreate(BaseModel):
    conversation_id: str
    message_content: str
    sender: str


class MyAiInput(BaseModel):
    user_input: str
